Check if divider not zeroΒΆ

def check_div(func):
    def check(a, b):
        if b == 0:
            print("Cant't divide by zero")
            return
        return func(a, b)
    return check
@check_div
def divide(a, b):
    return a // b

# divide = check_div(divide)

print(divide(10, 2))
print(divide(2, 0))
print(divide(15, 5))